import pandas as pd

def main():
    # 读取Excel文件
    excel_file = r'C:\Users\23248\PycharmProjects\stance\StanceDetectionLab\data\test_data.xlsx'  # 替换为你的Excel文件路径
    df = pd.read_excel(excel_file)

    # 定义简约背景的映射
    summary_backgrounds = {
        "新闻学教授怒骂张雪峰": "张雪峰直播中建议“打晕”报考新闻学的学生，引发新闻学教授不满，双方就专业价值与言论责任展开激烈争论。",
        "央视记者采访燕郊爆炸被驱离": "3月13日，燕郊爆燃事故后，记者现场采访因天然气泄漏风险被劝离，因沟通不当引发舆论质疑，指挥部致歉。",
        "姜萍数学竞赛违规事件": "2024年6月，姜萍入围阿里数学竞赛后被曝作弊，调查确认其老师违规，二人成绩取消，暴露赛制漏洞与舆论争议。",
        "胡锡进谈虐猫事件": "东南大学学生徐某因虐猫被举报，思想品德考核不合格，接连被南京大学、兰州大学拒录，引发社会争议。",
        "巴以冲突": "2024年，巴以冲突持续升级，以色列拒建巴勒斯坦国，没收西岸土地，加沙冲突已致超万名妇女死亡。"
    }

    # 添加新列“简约背景”
    df['简约背景'] = df['topic'].map(summary_backgrounds)

    # 保存修改后的Excel文件
    output_file = r'/StanceDetectionLab/data/test_data.xlsx'  # 替换为你想保存的文件名
    df.to_excel(output_file, index=False)

    print(f"文件已保存为 {output_file}")

if __name__ == '__main__':
    main()